/**
 * This program is free software, you can redistribute it and/or modify.
 * Copyright (c) 2025 Huawei Technologies Co., Ltd.
 * This file is a part of the CANN Open Software.
 * Licensed under CANN Open Software License Agreement Version 2.0 (the "License").
 * Please refer to the License for details. You may not use this file except in compliance with the License.
 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
 * See LICENSE in the root of the software repository for the full text of the License.
 */

#ifndef _SWI_GLU_GRAD_TILING_H_
#define _SWI_GLU_GRAD_TILING_H_

#include "kernel_tiling/kernel_tiling.h"

#define DT_BF16 bfloat16_t
#define ORIG_DTYPE_START DT_BF16
#define __CCE_UT_TEST__

#pragma pack(1)

struct SwiGluTilingData {
    uint32_t is32BAligned;
    uint32_t isDoubleBuffer;
    uint64_t rowLen;
    uint64_t colLen;
    uint32_t baseRowLen;
    uint32_t baseColLen;
};

#pragma pack()

#define CONVERT_TILING_DATA(tilingStruct, tilingDataPointer, tilingPointer)  \
    __ubuf__ tilingStruct* tilingDataPointer =                                 \
        reinterpret_cast<__ubuf__ tilingStruct*>((__ubuf__ uint8_t*)(tilingPointer));

#define INIT_TILING_DATA(tilingStruct, tilingDataPointer, tilingPointer)     \
    CONVERT_TILING_DATA(tilingStruct, tilingDataPointer, tilingPointer);

#define GET_TILING_DATA(tilingData, tilingPointer)                            \
    SwiGluTilingData tilingData;                                               \
    INIT_TILING_DATA(SwiGluTilingData, tilingDataPointer, tilingPointer);  \
    (tilingData).rowLen = tilingDataPointer->rowLen;                              \
    (tilingData).colLen = tilingDataPointer->colLen;                          \
    (tilingData).is32BAligned = tilingDataPointer->is32BAligned;            \
    (tilingData).isDoubleBuffer = tilingDataPointer->isDoubleBuffer;              \
    (tilingData).baseRowLen = tilingDataPointer->baseRowLen;            \
    (tilingData).baseColLen = tilingDataPointer->baseColLen;
#endif